import os
from selenium import webdriver
class Base():
    def setup(self):
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
        self.driver.implicitly_wait(5)
        ''' browser = os.getenv('browser')
                if browser == 'chrome':
                    self.driver = webdriver.Chrome()
                elif browser == 'firefox':
                    self.driver = webdriver.Firefox()
                elif browser == 'headless':
                    self.driver = webdriver.PhantomJS()
                else:
                    print('其他浏览器')'''
    def teardown(self):
        self.driver.quit()